package array;

/**
 * 
 * @Title: CandyArray.java 
 * @Package array 
 * @Description: 自定义一个数组：使其可以支持泛型的同时，可以动态扩缩容
 * @author CandyWall   
 * @date 2020年12月18日 下午7:43:44 
 * @version V1.0
 */
public class CandyArray_v3<E> {
	private E[] data;
	private int size;
	
	public CandyArray_v3() {
		this(10);
	}
	
	/**
	 * 传入数组的容量初始化Array
	 * @param capacity
	 */
	public CandyArray_v3(int capacity) {
		data = (E[]) new Object[capacity];
		size = 0;
	}
	
	/**
	 * 获取数组中的元素个数
	 * @return
	 */
	public int getSize() {
		return size;
	}
	
	/**
	 * 获取数组的容量
	 * @return
	 */
	public int getCapacity() {
		return data.length;
	}
	
	/**
	 * 判断数组是否为空
	 * @return
	 */
	public boolean isEmpty() {
		return size == 0;
	}
	
	/**
	 * 校验待操作的数组下标是否合法
	 */
	private void checkArrayIndexRange(int index) {
		if(index > size || index < 0) {
			throw new IllegalArgumentException("插入的index不合法！");
		}
	}
	
	/**
	 * 在数组末尾添加一个元素
	 * @param e
	 */
	public void addLast(E e) {
		add(size, e);
	}
	
	/**
	 * 在数组开头添加一个元素
	 * @param e
	 */
	public void addFirst(E e) {
		add(0, e);
	}
	
	/**
	 * 在第index个位置插入新元素
	 * @param index
	 * @param e
	 */
	public void add(int index, E e) {
		checkArrayIndexRange(index);

		if(size == data.length) {
			resize(data.length << 1);
		}
		
		//下标大于等于index的元素均后移一个单位
		for(int i = size; i > index; i--) {
			data[i] = data[i - 1];
		}
		
		data[index] = e;
		size++;
	}
	
	/**
	 * 数组扩容
	 */
	private void resize(int capacity) {
		E[] newArray = (E[]) new Object[capacity];
		System.arraycopy(data, 0, newArray, 0, data.length);
		data = newArray;
	}
	
	/**
	 * 获取index索引位置的元素
	 * @param index
	 * @return
	 */
	public E get(int index) {
		checkArrayIndexRange(index);
		
		return data[index];
	}
	
	/**
	 * 将index索引出的值改成e
	 * @param index
	 * @param e
	 */
	public void set(int index, E e) {
		checkArrayIndexRange(index);
		
		data[index] = e;
	}
	
	/**
	 * 查找数组是否有元素e
	 * @param e
	 */
	public boolean contains(E e) {
		for(int i = 0; i < size; i++) {
			if(data[i].equals(e)) {
				return true;
			}
		}
		return false;
	}
	
	/**
	 * 查找数组中元素e所在的索引，如果不存在就返回-1
	 * @param e
	 * @return
	 */
	public int find(E e) {
		for(int i = 0; i < size; i++) {
			if(data[i].equals(e)) {
				return i;
			}
		}
		return -1;
	}
	
	/**
	 * 从数组中删除index位置的元素，并返回删除的元素
	 * @return
	 */
	public E remove(int index) {
		checkArrayIndexRange(index);
		E ret = data[index];
		// 后面的元素往前挪填补空缺
		for(int i = index; i < size - 1; i++) {
			data[i] = data[i + 1];
		}
		size--;
		data[size] = null;
		
		// 如果数组中存放的元素是原来的四分之一，容量就减半
		if(size == data.length / 4) {
			resize(data.length >> 1);
		}
		return ret;
	}
	
	/**
	 * 从数组中删除第一个元素，并返回删除的元素
	 * @return
	 */
	public E removeFirst() {
		return remove(0);
	}
	
	/**
	 * 从数组中删除最后一个元素，并返回删除的元素
	 * @return
	 */
	public E removeLast() {
		return remove(size - 1);
	}
	
	/**
	 * 从数组中删除元素e
	 * @param e
	 */
	public void removeElement(E e) {
		int index = find(e);
		if(index != -1) {
			remove(index);
		}
	}

	@Override
	public String toString() {
		StringBuilder sb = new StringBuilder();
		sb.append(String.format("CandyArray: size = %d, capacity = %d, content: [", size, data.length));
		/*int i = 0;
		for(; i < size - 1; i++) {
			sb.append(data[i] + ", ");
		}
		sb.append(data[i] + "]");
		return sb.toString();*/
        for (int i = 0; ; i++) {
            sb.append(data[i]);
            if (i == size - 1)
                return sb.append(']').toString();
            sb.append(", ");
        }
	}
	
	public static void main(String[] args) {
		CandyArray_v3<Integer> arr = new CandyArray_v3(10);
		for(int i = 0; i < 10; i++) {
			arr.addLast(i);
		}
		System.out.println(arr);
		arr.add(8, 100);
		System.out.println(arr);
		arr.set(2, 1000);
		System.out.println(arr);
		arr.removeLast();
		System.out.println(arr);
		arr.removeElement(1000);
		System.out.println(arr);
	}
}
